"""
项目：Ai-Engineer自适应工程设计自动化插件
作者：刘昊轩、李佳音、刘伟伟、杨明久（排序不分先后）
©版权保护
"""


class OptimizationAdvisor:
    def __init__(self, ai_client, knowledge_base):
        self.ai = ai_client
        self.knowledge = knowledge_base

    def analyze_design(self, design_data):
        """执行设计优化分析"""
        prompt = f"""
        根据以下设计参数进行优化建议：
        材料：{design_data['material']}
        总重量：{design_data['weight']} kg
        制造成本：{design_data['cost']} 元

        请考虑：
        1. 材料强度是否充足
        2. 可替代的制造工艺
        3. 成本优化方案
        """

        return self.ai.generate(prompt)

    def calculate_cost(self, design):
        """计算制造成本"""
        material_cost = design['volume'] * self.knowledge.get_material_property(
            design['material'], 'cost_per_kg')

        process_cost = design['manufacturing_time'] * self.knowledge.MANUFACTURING[
            design['process']]['cost_per_hour']

        return material_cost + process_cost